import django_filters
from django.db.models import Q

from .models import DanceCompany, CityPage


class DanceCompanyFilter(django_filters.rest_framework.FilterSet):

    class Meta:
        model = DanceCompany
        fields = ['state']

class CityPageFilter(django_filters.rest_framework.FilterSet):

    class Meta:
        model = CityPage
        fields = ['pici']